package process2

import (
	"fmt"
)

// 因为 UserMgr 实例在服务器端有且只有一个，并且很多的地方，都会使用到，所以将其定义为全局变量
var (
	userMgr *UserMgr
)

type UserMgr struct {
	onlineUsers map[int]*UserProcess
}

// 完成对 userMgr 初始化工作
func init() {
	userMgr = &UserMgr{
		onlineUsers: make(map[int]*UserProcess, 1024),
	}
}

// 完成对 onlineUsers 添加
func (this *UserMgr) AddOnlineUser(up *UserProcess) {
	this.onlineUsers[up.UserId] = up
}

// 删除
func (this *UserMgr) DelOnlineUser(userId int) {
	delete(this.onlineUsers, userId)
}

// 返回当前所有在线的用户
func (this *UserMgr) GetAllOnlineUser() map[int]*UserProcess {
	return this.onlineUsers
}

// 根据 id 返回对应的用户
func (this *UserMgr) GetOnlineUserById(userId int) (up *UserProcess, err error) {
	// 从 map 取出一个值，带检测方式
	up, ok := this.onlineUsers[userId]
	if !ok { // 要查找的这个用户，当前不在线。
		err = fmt.Errorf("用户%d 不存在", userId)
		return
	}
	return
}
